Train 1D CNN

1D Convolution을 기본 구성 요소로 하는 EEG classifier를 학습해보는 노트북.


환경 구성


Data Filtering by Diagnosis

Non-Vascular Dementia, Non-Vascular MCI, Normal


Configure the Train, Validation, and Test Splits

Split the filtered dataset and shuffle them

Wrap the splitted data using PyTorch Dataset

Data loader test

Train, validation, test dataloaders


Some useful functions for training


Train 1D CNN models

1D CNN: M5-like model

1D CNN: M5-like model without age

1D CNN: ResNet-like model

1D CNN: ResNet-like model without age